Skip to content

Fix recursive search in Client.get_items #799

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 9 commits into
base: main
Choose a base branch
from

Conversation

mishaschwartz
Copy link

@mishaschwartz mishaschwartz commented May 14, 2025

Related Issue(s):

Description:

  • runs a non-recursive search when using the recursive argument is False
  • updates tests

PR Checklist:

  • Code is formatted
  • Tests pass
  • Changes are added to CHANGELOG.md

@jsignell jsignell self-requested a review May 15, 2025 13:07
@jsignell jsignell self-assigned this May 15, 2025
Copy link
Member

@jsignell jsignell left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I have a few suggestions, but thank you so much for opening this PR! I think it'll be a real improvement.

Comment on lines 465 to 467
except APIError:
child_catalogs = [catalog for catalog, _, _ in self.walk()]
search = self.search(ids=ids, collections=[self, *child_catalogs])
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This seems like it would be pretty easy to do accidentally. I think I'd prefer to just let the error raise and make it a little harder to get every single item in planetary computer for instance.

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

My concern is that without something like this, many functions that call get_items simply don't work for planetary computer or similar APIs that enforce this required argument. This includes:

  • Client.get_all_items,
  • Client.walk,
  • Client.validate_all,
  • Client.describe,
  • Client.make_all_asset_hrefs_relative,
  • Client.make_all_asset_hrefs_absolute

Note that the spec doesn't say one way or another that these arguments must be optional so I'm guessing that planetary computer's API is still spec compliant technically. However, the examples show that a search without collections should be supported so I don't really know one way or the other how to interpret that:

https://github.com/radiantearth/stac-api-spec/blob/604ade6158de15b8ab068320ca41e25e2bf0e116/item-search/examples.md?plain=1#L27

Otherwise the only way to make this work for APIs like planetary computer is to override the Client class like:

import pystac_client

class Client(pystac_client.Client):
    def search(self, *args, **kwargs):
        if kwargs["collections"] is None:
            kwargs["collections"] = [self.id *[catalog.id for catalog, _, _ in self.walk()]]
        return super().search(*args, **kwargs)

pystac_client.client.Client = Client  # so that sub-catalogs also use the updated search method 

If that's the approach we want to go with that's fine, but maybe we should document this workaround in case users want to interact with planetary computer.

What do you think?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for taking the time to write that all up! I think as long as a clear error surfaces it is fine to have those methods fail on Planetary Computer. Requiring collections is not technically compliant with the spec, so I think it is better to not bake in special handling for this scenario especially since it is likely to result in a surprising user experience (setting collections to include every collection might be very very slow).

@jsignell jsignell removed their assignment May 15, 2025
@mishaschwartz mishaschwartz requested a review from jsignell May 20, 2025 18:24
@mishaschwartz
Copy link
Author

@jsignell I realize that the default value for recursive should actually be False so that it matches the signature for Catalog.get_items (see: https://github.com/mishaschwartz/pystac/blob/7621455aef4813bfe2da571d88484d20bf616122/pystac/catalog.py#L538).

This is necessary because all of the pystac code assumes that the default value is False so many methods that pystac_client.Client inherits from pystac.Catalog will return unexpected results.

For example: if you have a catalog which contains subcatalogs, then Client.walk will yield items in subcatalogs multiple times (one for each ancestor catalog) since get_items is recursive by default.

I would like to make the default False but I realize that that is a breaking change for the pystac_client's public interface. Is that a change that you're comfortable with me making in this PR?

@jsignell
Copy link
Member

@jsignell I realize that the default value for recursive should actually be False so that it matches the signature for Catalog.get_items (see: https://github.com/mishaschwartz/pystac/blob/7621455aef4813bfe2da571d88484d20bf616122/pystac/catalog.py#L538).

This is necessary because all of the pystac code assumes that the default value is False so many methods that pystac_client.Client inherits from pystac.Catalog will return unexpected results.

For example: if you have a catalog which contains subcatalogs, then Client.walk will yield items in subcatalogs multiple times (one for each ancestor catalog) since get_items is recursive by default.

I would like to make the default False but I realize that that is a breaking change for the pystac_client's public interface. Is that a change that you're comfortable with me making in this PR?

What if we just leave the default as None? That way this PR is purely additive -- it adds the option to set recursive=False and otherwise changes nothing.

@codecov-commenter
Copy link

codecov-commenter commented May 21, 2025

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 94.22%. Comparing base (21435b0) to head (8a268d2).
Report is 123 commits behind head on main.

Additional details and impacted files
@@            Coverage Diff             @@
##             main     #799      +/-   ##
==========================================
+ Coverage   93.43%   94.22%   +0.79%     
==========================================
  Files          13       15       +2     
  Lines         990     1213     +223     
==========================================
+ Hits          925     1143     +218     
- Misses         65       70       +5     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@mishaschwartz
Copy link
Author

mishaschwartz commented May 21, 2025

What if we just leave the default as None? That way this PR is purely additive -- it adds the option to set recursive=False and otherwise changes nothing.

If we leave the default as None that's fine but we would need to change the way that the code treats a None value. Right now None is treated the same as True, we would need the None to be treated the same as False.

Either way, it changes the semantics of Client.get_items() without any arguments (which is the way that it is called in various pystac.Catalog methods).

@jsignell
Copy link
Member

If we leave the default as None that's fine but we would need to change the way that the code treats a None value. Right now None is treated the same as True, we would need the None to be treated the same as False.

Either way, it changes the semantics of Client.get_items() without any arguments (which is the way that it is called in various pystac.Catalog methods).

Sorry why not just let None be treated as True?

@mishaschwartz
Copy link
Author

Sorry why not just let None be treated as True?

OK here's an example:

  • pystac.Catalog.walk calls self.get_items() (Note: without any arguments so the default values are used!)
  • pystac.Catalog.get_items has recursive = False by default
  • this means that pystac.Catalog.walk is trying to get items non-recursively
  • pystac_client.Client.get_items has recursive = None by default which is the same as recursive = True
  • pystac_client.Client.walk is inherited from pystac.Catalog but will call pystac_client.Client.get_items
  • Therefore pystac_client.Client.walk will recursively call get_items even though we want it to call it non-recursively

A concrete example:

Let's say you have the following structure:

  • catalog1

    • catalog2
      • item1
  • pystac.Catalog.walk will tell you that item1 is a child of catalog2 but not catalog1

  • pystac_client.Client.walk will tell you that item1 is a child of catalog1 and catalog2

@jsignell
Copy link
Member

Thank you for spelling it out for me. This inheritance model is not pretty. I think I feel ok with changing the default to recursive=False as long as we use recursive=True in get_all_items

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Client.get_items has surprising recursive behaviour when using the /search endpoint
3 participants